# encoding: utf-8

full_data = "./datas/pretrain_data.txt"
demo_data = "./datas/pretrain_demo_data.txt"

size = 2000
index = 0
with open(full_data, "r", encoding="utf-8") as f:
    with open(demo_data, "w", encoding="utf-8") as ff:
        for line in f:
            ff.write(line)
            index += 1
            if index > size:
                break